3 axis in python